43ca78fea0115e05c8d14626f470010e2f7334c3,tez-runtime-library/src/test/java/org/apache/tez/dag/library/vertexmanager/TestFairShuffleVertexManager.java,TestFairShuffleVertexManager,testFairSchedulingWithPartitionStats,#,146
Before Change
Assert.assertEquals(2, edgeManager.getNumDestinationTaskPhysicalInputs(2));
for (int sourceTaskIndex = 1; sourceTaskIndex < 3; sourceTaskIndex++) {
for (int j = 0; j < 2; j++) {
routeMetadata = (j == 0) ?
edgeManager.routeCompositeDataMovementEventToDestination(
sourceTaskIndex, 2) :
edgeManager.routeInputSourceTaskFailedEventToDestination(
sourceTaskIndex, 2);
Assert.assertEquals(1, routeMetadata.getNumEvents());
if (j == 0) {
Assert.assertEquals(2, routeMetadata.getSourceIndices()[0]);
}
Assert.assertEquals(sourceTaskIndex - 1,
routeMetadata.getTargetIndices()[0]);
After Change
Assert.assertEquals(2, edgeManager.getNumDestinationTaskPhysicalInputs(2));
for (int sourceTaskIndex = 1; sourceTaskIndex < 3; sourceTaskIndex++) {
for (int j = 0; j < 2; j++) {
if (j == 0) {
EdgeManagerPluginOnDemand.CompositeEventRouteMetadata routeMetadata =
edgeManager.routeCompositeDataMovementEventToDestination(sourceTaskIndex, 2);
Assert.assertEquals(1, routeMetadata.getCount());
Assert.assertEquals(2, routeMetadata.getSource());
Assert.assertEquals(sourceTaskIndex - 1, routeMetadata.getTarget());
} else {
EdgeManagerPluginOnDemand.EventRouteMetadata routeMetadata =
edgeManager.routeInputSourceTaskFailedEventToDestination(sourceTaskIndex, 2);
Assert.assertEquals(1, routeMetadata.getNumEvents());
Assert.assertEquals(sourceTaskIndex - 1, routeMetadata.getTargetIndices()[0]);
}